Train Text Classification Model and Save to Mlflow

I) Train Model

Data Prep

Feature Engineering

Create tokens in prepro step, hence pass tokens to tfidf vect.

Training

Evalution

Skip validation set performance evaluation. Short check on training data only for functionality test.

II) Save to Mlflow

Use custom pyfunc implementation due to the nltk dependency

Save Local

Save to MLflow